Unlocking Token Rewards via Training-Free Reward Attribution
P2T:免训练将过程奖励分解为 Token 级奖励
🧭 RL 奖励归因 · Token-level reward ★ 深度解读
Chinese University of Hong Kong · Hong Kong University of Science and Technology · University of Hong Kong
提出训练免费的 token 级奖励归因方法,用梯度一阶泰勒近似从 PRM 分解过程奖励,加速 RL 训练并提升推理表现
📌 工作参考价值(4 点)
- 若业务中已有可微分的奖励模型(如用户满意度预测、转化模型、Critic、视觉质量模型),可直接用其梯度对生成序列做 token/action 级归因,无需训练新 reward model;公式简单:token influence ≈ ∇_{e_i} R(E)·(e_i - e_null),一个 backward pass 拿到全序列。
- 将粗粒度奖励作为 baseline 再叠加归一化影响权重,而非直接用影响分数作为奖励,能显著降低梯度噪声;η 从 0.25 到 1.0 都稳,业务上无需精细调参。选择 null token 用模型 padding token 比 EOS/zero embedding 更稳。
- 训练 Agent/生成式推荐策略时,如果只有最终 outcome label(点击/转化/任务成功),可接入现成 PRM 或可微分 reward model,用 P2T 方式分解到每个 action/token,加速 RL 收敛;论文显示达到同等效果可少 3-4 倍训练步数。
- 注意 PRM 质量决定上限;若 PRM 在目标域不可靠,收益有限,应先验证 PRM 在该域判断准确,且 α 可从小增大。